package com.atguigu.network;

import org.junit.Test;

import java.io.*;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;

/**
 * @author yihu
 * @create 2021-06-15 11:07
 * @email:1057202547@qq.com
 * @dec:实现TCP的网络编程
 *  * 例子1：客户端发送信息给服务端，服务端将数据显示在控制台上
 */
public class TCPTest1 {
    //客户端
    @Test
    public void client()  {
        Socket socket = null;
        OutputStream os = null;
        try {
            InetAddress inetAddress = InetAddress.getByName("127.0.0.1");
            //1.创建一个Soket对象，指明服务器端的ip 和端口号（服务器端的ip和端口号)
            socket = new Socket(inetAddress,8899);
            //2.获取一个输出流 输出数据
            os = socket.getOutputStream();
            //3.写出数据的超作
            os.write("您好我是客户端mm".getBytes());
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            //4.关闭资源
            if(os!=null){
                try {
                    os.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if(socket!=null){
                try {
                    socket.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    //服务端
    @Test
    public void server() {
        ServerSocket ss = null;
        Socket socket = null;
        BufferedInputStream bis = null;
        ByteArrayOutputStream baos = null;
        try {
            //1.创建服务起端的ServerSocket 指明自己的端口
            ss = new ServerSocket(8899);
            //2.调用accept()表示接收来自于客户端的socket
            socket = ss.accept();
            //3.获取客户端socket 的输入流
            InputStream is = socket.getInputStream();
            bis = new BufferedInputStream(is);
            //字节还原字符串可能会出先乱码
            //byte[] buffer = new byte[1024];
            //int len;
            //while ((len = bis.read(buffer))!=-1){
            //  String str = new String(buffer,0,len);
            //  System.out.println(str);
            //}
            //将读出的字节先写到baos 里面 在一起进行还原 （底层是一个 byte buf[] 不断的扩容，）
            baos = new ByteArrayOutputStream();
            byte[] buffer = new byte[1024];
            int len;
            while ((len = bis.read(buffer))!=-1){
                //4.读取输入流中的数据
                baos.write(buffer,0,len);
            }
            String str = new String(baos.toByteArray());
            System.out.println(str);
            System.out.println( "收到了来自于:" + socket.getInetAddress().getHostAddress());
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if(bis!=null){
                //关闭资源
                try {
                    bis.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if(baos!=null){
                //关闭资源
                try {
                    baos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if(socket!=null){
                //关闭资源
                try {
                    socket.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if(ss!=null){
                //关闭资源
                try {
                    ss.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }
}
